import os

class Experiment_config(object):

    def __init__(self):
        self.cuda_device = '0'

        self.arch = 'resnet50'
        self.batch_size = 64
        self.combine_trainval = True
        self.data_dir = '/home/zzz/wa_reid/market_dataset'
        self.dataset = 'market'
        self.dropout = 0.5
        self.epochs = 100
        self.evaluate = True

        self.dim_final_features = 2048
        self.dim_triplet_features = 256
        self.features = 256
        self.height = 384

        self.lr = 0.1
        self.momentum = 0.9
        self.print_freq = 1

        self.seed = 1
        self.split = 0
        self.step_size = 40
        self.weight_decay = 0.0005
        self.width = 128
        self.workers = 4

        self.param_triplet_loss = 0.5

        self.loss_type = ['cross_entropy', 'triplet']
        self.loss_option = 1

        # self.resume = ''
        # self.logs_dir = '/home/ld/reid_related/{}_baseline_2.28'.format(self.loss_type[self.loss_option])
        self.logs_dir = '/home/ld/reid_related/{}_baseline'.format(self.loss_type[self.loss_option]) +\
                        '_param_triplet_loss_{}'.format(self.param_triplet_loss)
        self.resume = self.logs_dir + '/checkpoint.pth.tar'
